Fine tune Vision Transformers